package src.DisciplesPlan.链表;

/**
 * 给你一个链表，每 k 个节点一组进行翻转，请你返回翻转后的链表。

 k 是一个正整数，它的值小于或等于链表的长度。

 如果节点总数不是 k 的整数倍，那么请将最后剩余的节点保持原有顺序。

 进阶：

 你可以设计一个只使用常数额外空间的算法来解决此问题吗？
 你不能只是单纯的改变节点内部的值，而是需要实际进行节点交换。
  

 示例 1：


 输入：head = [1,2,3,4,5], k = 2
 输出：[2,1,4,3,5]
 示例 2：


 输入：head = [1,2,3,4,5], k = 3
 输出：[3,2,1,4,5]
 示例 3：

 输入：head = [1,2,3,4,5], k = 1
 输出：[1,2,3,4,5]
 示例 4：

 输入：head = [1], k = 1
 输出：[1]
 */

public class _25K个一组翻转链表 {
    public static void main(String[] args) {
        ListNode head = TestLinkedList.generateLinkedList(2);
        TestLinkedList.showLinkedList(head);

        ListNode p = reverseKGroup(head, 2);

        TestLinkedList.showLinkedList(p);
    }
    public static ListNode reverseKGroup(ListNode head, int k) {
        // 虚拟头结点(便于反转)
        ListNode virNode = new ListNode(0);
        // 连接到原链表头结点上
        virNode.next = head;
        // 记录每一次的要反转的区域的前一个结点
        ListNode p = virNode;
        // 记录每一次要反转的区域的头结点
        ListNode q = p.next;
        while ( (p.next = correctN(q, k)) != q) {
            p = q;
            q = p.next;
        }
        return virNode.next;
    }

    // 判断剩下的链表是否足够有n个结点
    public static ListNode correctN(ListNode head, int n) {
        // 辅助指针
        ListNode p = head;
        // 记录长度
        int count = n;
        while (n != 0 && p != null) {
            n--;
            p = p.next;
        }
        // 此时要分两种情况讨论
        // 一:步数刚好等于剩下的结点数，此时指针刚好位于null，此时也需要反转
        if (p == null && n == 0) {
            return reverseN(head, count);
        }
        // 二:后移n步的过程中因为p指针置空才退出循环，此时n不为空
        else if (p == null){
            return head;
        }
        // 如果够n个结点，则反转并返回其头结点
        return reverseN(head, count);
    }

    // 反转前n个结点
    public static ListNode reverseN(ListNode head, int n) {
        // 如果遍历到最后一个结点则直接返回
        if (n == 1) {
            return head;
        }
        // 递归依次遍历到最后两个结点
        ListNode tail = head.next;
        ListNode p = reverseN(head.next, n - 1);
        // 反转最后两个结点
        head.next = tail.next;
        tail.next = head;
        // 返回反转后的链表的头结点，也即原链表的最后一个结点
        return p;
    }
}
